import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketException;

// 服务器
public class UdpEchoServer {
    // 定义一个 socket 对象
    // 网络通信, 必须要 socket 对象
    private DatagramSocket datagramSocket = null;

    // 构造 socket, 服务端需要绑定端口
    // 关联/绑定一个端口, 不一定可以成功
    // 如果这个端口被别的进程占用了, 就会出错
    // 同一个主机上, 同一个端口, 同一时刻, 只能被一个进程绑定
    public UdpEchoServer(int port) throws SocketException {
        datagramSocket = new DatagramSocket(port);
    }

    // 执行服务器的主要逻辑
    public void start() throws IOException {
        System.out.println("服务器启动!!");
        while (true) {
            // 1. 接收请求, 并解析
            // 用来接收请求数据, 一个饭盒, 请求数据包
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096], 4096);

            // 接收请求数据, 将饭盒给 receive, 给我们填充数据
            // 如果没有客户端请求, 这个 receive 会阻塞等待
            datagramSocket.receive(requestPacket);
            // 将这个数据转成 String , 方便我们解析
            String request = new String(requestPacket.getData(), 0, requestPacket.getLength());

            // 2. 根据请求计算响应(省略这个步骤)
            String response = process(request);

            // 3. 把响应结果返回给客户端
            // 需要明确传给哪个客户端, getSocketAddress 可以获取到客户端的 ip 和 端口
            // 响应数据包
            DatagramPacket responsePacket = new DatagramPacket(response.getBytes(), response.getBytes().length,
                    requestPacket.getSocketAddress());

            datagramSocket.send(responsePacket);

            // 打印日志, ip, 端口, 请求数据, 响应数据
            System.out.printf("[%s:%d] req: %s, resp: %s\n", requestPacket.getAddress().toString(),
                    requestPacket.getPort(), request, response);
        }
    }


    // 根据客户端的请求, 计算服务器响应
    // 我们写的是 回显程序, 请求是啥, 响应就是啥
    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        UdpEchoServer udpEchoServer = new UdpEchoServer(6666);
        udpEchoServer.start();
    }
}
